# pip install zhipuai 请先在终端进行安装

import zhipuai

zhipuai.api_key = "41805a5412196acfdd9d63a905f9039a.Z0TWy2173NoP1ZkI"
response = zhipuai.model_api.sse_invoke(
    model="chatglm_lite",
    prompt= [{"role":"user","content":"请你作为一名新闻学专业的教授，帮助你的学生润色论文，以下是需要润色的内容：\n随着青年世代逐渐成熟， Z世代正逐步登上社会发展的大舞台，成为职场的新生主力军。进入职场的第一步就是进行职业选择，对于 Z 世代来说，职业选择可能面临着许多不同的挑战和机遇 。一方面是，相比起前几代人，Z世代成长于鼓励创新和自由民主的社会大环境下，他们有着更加开放和自由的思维方式，更加注重自我实现和价值观的匹配；另一方面是他们深受数字化的影响，在互联网和社交媒体的熏陶下，他们对多元文化有更深的认同感。因此，他们在进行职业选择时所关注的维度也更加多元化，同时也更加关注自身，进行职业选择的标尺也更为宽幅、更为多变，择业不再仅受单一因素左右，而是从多种角度考虑。"}],
    temperature= 0.9,
    top_p= 0.7,
    incremental=True
)

for event in response.events():
    if event.event == "add":
        print(event.data, end="")
    elif event.event == "error" or event.event == "interrupted":
        print(event.data, end="")
    elif event.event == "finish":
        print(event.data)
        print(event.meta, end="")
    else:
        print(event.data, end="")